理解Swift中map 和 flatMap对集合的作用

map和flatMap是函数式编程中常见的概念,python等语言中都有。借助于 map和flapMap 函数能够很轻易地将数组转换成另外一个新数组。

map函数能够被数组调用,它接受一个闭包作为参数,作用于数组中的每个元素。闭包返回一个变换后的元素,接着将所有这些变换后的元素组成一个新的数组。
简单的说, map就是映射函数,把一个集合映射成另一个集合。

Swift的flatMap不容易理解,flatMap很像map函数,但是它摒弃了那些值为nil的元素。
flatMap是在处理一个容器而不是数组,这样可以更好地理解它。

为了处理可选类型,flatMap被重载过。它会接受一个可选类型的数组并返回一个拆包过的且没有nil值的可选类型组成的数组。
下面是flatMap的定义

1
2
3
4
5
6
7
8
9
10
11
12
13
extension SequenceType {

/// Return an `Array` containing concatenated results of mapping `transform`
/// over `self`.
func flatMap<S : SequenceType>(@noescape transform: (Self.Generator.Element) -> S) -> [S.Generator.Element]
}

extension SequenceType {

/// Return an `Array` containing the non-nil results of mapping `transform`
/// over `self`.
func flatMap<T>(@noescape transform: (Self.Generator.Element) -> T?) -> [T]
}

参考下面的示例代码:

1
2
3
4
5
6
7
8
9
let test1 = (1...7).map{ (i : Int) -> String?  in
return i > 1 ? String("sss \(i)") : nil
}
print(test1)

let test2 = (1...7).flatMap{ (i : Int) -> String? in
return i > 1 ? String("sss \(i)") : nil
}
print(test2)

执行结果
[nil, Optional(“sss 2”), Optional(“sss 3”), Optional(“sss 4”), Optional(“sss 5”), Optional(“sss 6”), Optional(“sss 7”)]
[“sss 2”, “sss 3”, “sss 4”, “sss 5”, “sss 6”, “sss 7”]

参考文章

http://swift.gg/2015/08/06/swift-2-flatmap/

https://segmentfault.com/a/1190000004050907

Contents
,